﻿
// FileTranslateServerDlg.cpp: 实现文件
//

#include "pch.h"
#include "framework.h"
#include "FileTranslateServer.h"
#include "FileTranslateServerDlg.h"
#include "afxdialogex.h"
#include "CListenSocket.h"
#include "CClientSocket.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

//定义端口号10086
#define PORT 10087

// CFileTranslateServerDlg 对话框



CFileTranslateServerDlg::CFileTranslateServerDlg(CWnd* pParent /*=nullptr*/)
	: CDialogEx(IDD_FILETRANSLATESERVER_DIALOG, pParent)
	, m_strFilePath(_T(""))
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

//数据交换
void CFileTranslateServerDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_LIST, m_list);
	DDX_Text(pDX, IDC_FILE_PATH, m_strFilePath);
}

BEGIN_MESSAGE_MAP(CFileTranslateServerDlg, CDialogEx)
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BTN_BROWSER, &CFileTranslateServerDlg::OnBnClickedBtnBrowser)
	ON_BN_CLICKED(IDC_BTN_SEND, &CFileTranslateServerDlg::OnBnClickedBtnSend)
END_MESSAGE_MAP()


// CFileTranslateServerDlg 消息处理程序

BOOL CFileTranslateServerDlg::OnInitDialog()
{
	CDialogEx::OnInitDialog();

	// 设置此对话框的图标。  当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO: 在此添加额外的初始化代码
	m_list.SetExtendedStyle(LVS_EX_FULLROWSELECT);
	m_list.InsertColumn(0, _T("IP地址"), LVCFMT_LEFT, 250);
	m_list.InsertColumn(1, _T("端口"), LVCFMT_CENTER, 100);

	//创建一个对象
	m_pListenSocket = new CListenSocket;
	//创建套接字
	CString strMsg;
	if (FALSE == m_pListenSocket->Create(PORT, SOCK_STREAM))
	{
		strMsg.Format(_T("创建套接字失败，错误编号:%d"), GetLastError());
		MessageBox(strMsg, _T("温馨提示"), MB_OK);
		delete m_pListenSocket;
		m_pListenSocket = NULL;
		//关闭对话框
		EndDialog(IDOK);
		return TRUE;
	}

	//将套接字设置为监听模式
	if (FALSE == m_pListenSocket->Listen())
	{
		strMsg.Format(_T("监听失败，错误编号:%d"), GetLastError());
		MessageBox(strMsg, _T("温馨提示"), MB_OK);
		delete m_pListenSocket;
		m_pListenSocket = NULL;
		//关闭对话框
		EndDialog(IDOK);
		return TRUE;
	}



	return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。  对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CFileTranslateServerDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作区矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialogEx::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CFileTranslateServerDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}


//浏览文件
void CFileTranslateServerDlg::OnBnClickedBtnBrowser()
{
	CFileDialog dlg(TRUE);
	if (IDCANCEL == dlg.DoModal())//弹出
	{
		return;
	}

	//获取文件路径并写在文本框里面
	m_strFilePath = dlg.GetPathName();

	//防止刷新导致字符串被清空
	UpdateData(FALSE);
}

//发送
void CFileTranslateServerDlg::OnBnClickedBtnSend()
{
	UpdateData(TRUE);

	if (m_strFilePath.IsEmpty())
	{
		MessageBox(_T("请选择需要发送的文件"), _T("温馨提示"), MB_OK);
		return;
	}

	int nSel = m_list.GetSelectionMark();
	if (nSel < 0)
	{
		MessageBox(_T("请选择需要发送的对象"), _T("温馨提示"), MB_OK);
		return;
	}

	//获取套接字
	CClientSocket* pSocket = (CClientSocket*)m_list.GetItemData(nSel);

	//获取文件属性
	WIN32_FIND_DATA wfd;
	HANDLE hFinder = FindFirstFile(m_strFilePath, &wfd);
	FindClose(hFinder);

	//先发送文件属性
	pSocket->Send(&wfd, sizeof(wfd));

	CFile file;
	//以读的方式打开文件
	if (FALSE == file.Open(m_strFilePath, CFile::modeRead | CFile::typeBinary))
	{
		MessageBox(_T("打开文件失败"), _T("温馨提示"), MB_OK);
		return;
	}

	//读取文件并循环发送
	DWORD dwReadCount = 0;
	while (dwReadCount < wfd.nFileSizeLow)
	{
		char* buffer = new char[1025];
		int nRead = file.Read(buffer, 1024);
		buffer[nRead] = '\0';

		//发送
		pSocket->Send(buffer, nRead);
		dwReadCount += nRead;
		delete[] buffer;
	}


	//关闭文件
	file.Close();

	MessageBox(_T("发送数据完毕"), _T("温馨提示"), MB_OK);


}
